tensor([[0.2900, 0.2900, 0.2900, 0.2900, 0.2900],
        [0.7100, 0.7100, 0.7100, 0.7100, 0.7100]], grad_fn=<RoundBackward1>)